package 高效制胜._05;

import java.util.*;

/**
 * @author yzh
 * @version 1.0
 * @date 2021/9/11 12:12
 * 下一个更大元素 I
 * 算法：单调栈
 * 元素唯一
 * 从左往右遍历 nums2，如果栈为空，将元素直接放入栈中
 * 如果不为空，则不断比较当前元素和栈顶元素，如果当前元素大于栈顶元素
 * 就将栈顶元素弹出，说明比它大的数是当前元素，记录到 map 中
 */
public class _496 {

    public int[] nextGreaterElement_violence(int[] nums1, int[] nums2) {
        int[] ans = new int[nums1.length];
        Arrays.fill(ans, -1);
        for (int i = 0; i < nums1.length; i++) {
            boolean flag = false;
            for (int j = 0; j < nums2.length; j++) {
                if (nums2[j] == nums1[i]) flag = true;
                if (flag && nums2[j] > nums1[i]) {
                    ans[i] = nums2[j];
                    break;
                }
            }
        }
        return ans;
    }

    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        Map<Integer, Integer> map = new HashMap<>(nums2.length);
        Deque<Integer> stack = new LinkedList<>();
        for (int num : nums2) {
            if (stack.isEmpty()) stack.push(num);
            while (!stack.isEmpty() && num > stack.peek()) map.put(stack.pop(), num);
            stack.push(num);
        }
        int[] ans = new int[nums1.length];
        for (int i = 0; i < nums1.length; i++) ans[i] = map.getOrDefault(nums1[i], -1);
        return ans;
    }

}
